package com.jxinternet.platform.services.util;

import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Arrays;


/**
 * Created by gold-jx on 2017/6/3.
 */
public class MD5Util {


    //16进制需要的字符串数据
    private static final String HEX_NUMS_STR = "0123456789ABCDEF";


    //随机字符串字节数长度
    private static final Integer BYTE_LENGTH = 18;


    /**
     * 将16进制字符串转换成字节数组
     *
     * @param hex
     * @return
     */
    public static byte[] hexStringToByte(String hex) {
        int len = (hex.length() / 2);
        byte[] result = new byte[len];
        char[] hexChars = hex.toCharArray();
        for (int i = 0; i < len; i++) {
            int pos = i * 2;
            result[i] = (byte) (HEX_NUMS_STR.indexOf(hexChars[pos]) << 4 | HEX_NUMS_STR.indexOf(hexChars[pos + 1]));
        }
        return result;
    }


    /**
     * 将指定byte数组转换成16进制字符串
     *
     * @param b
     * @return
     */
    public static String byteToHexString(byte[] b) {
        StringBuffer hexString = new StringBuffer();
        for (int i = 0; i < b.length; i++) {
            String hex = Integer.toHexString(b[i] & 0xFF);
            if (hex.length() == 1) {
                hex = '0' + hex;
            }
            hexString.append(hex.toUpperCase());
        }
        return hexString.toString();
    }

    public static String getMD5to16(String plainText) {

        String result = "";

        try {
            //生成实现指定摘要算法的 MessageDigest 对象。
            MessageDigest md = MessageDigest.getInstance("MD5");
            //使用指定的字节数组更新摘要。
            md.update(plainText.getBytes());
            //通过执行诸如填充之类的最终操作完成哈希计算。
            byte[] b = md.digest();
            //生成的md5换成16进制字符串
            result = byteToHexString(b);
            //System.out.println("temp16位: " + result);

        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }


    /**
     * 获得加密后的16进制形式口令
     *
     * @param password
     * @return
     * @throws NoSuchAlgorithmException
     * @throws UnsupportedEncodingException
     */
    public static String getEncryptedPwd(String password)
            throws NoSuchAlgorithmException, UnsupportedEncodingException {
        // 声明加密后的口令数组变量
        byte[] pwd = null;
        // 随机数生成器
        SecureRandom random = new SecureRandom();
        // 声明随机数组变量
        byte[] randomByte = new byte[BYTE_LENGTH];
        // 将随机数放入随机数组变量中
        random.nextBytes(randomByte);


        // 声明消息摘要对象
        MessageDigest md = null;
        // 创建消息摘要
        md = MessageDigest.getInstance("MD5");
        // 将原数据传入消息摘要对象
        md.update(randomByte);
        // 将口令的数据传给消息摘要对象
        md.update(password.getBytes("UTF-8"));
        // 获得消息摘要的字节数组
        byte[] digest = md.digest();


        // 因为要在口令的字节数组中存放随机数组密文，所以加上随机数组的字节长度
        pwd = new byte[digest.length + BYTE_LENGTH];
        // 将原的字节拷贝到生成的加密口令字节数组的前相应该长度BYTE_LENGTH个字节，以便在验证口令时取出随机数组
        System.arraycopy(randomByte, 0, pwd, 0, BYTE_LENGTH);
        // 将消息摘要拷贝到加密口令字节数组从第BYTE_LENGTH个字节开始的字节
        System.arraycopy(digest, 0, pwd, BYTE_LENGTH, digest.length);
        // 将字节数组格式加密后的口令转化为16进制字符串格式的口令


        return byteToHexString(pwd);
    }


    /**
     * 验证口令是否合法
     *
     * @param password
     * @param passwordInDb
     * @return
     * @throws NoSuchAlgorithmException
     * @throws UnsupportedEncodingException
     */
    public static boolean validPassword(String password, String passwordInDb)throws NoSuchAlgorithmException, UnsupportedEncodingException {
        // 将16进制字符串格式口令转换成字节数组
        byte[] pwdInDb = hexStringToByte(passwordInDb);
        // 声明一个随机数组变量
        byte[] randomByte = new byte[BYTE_LENGTH];
        // 将随机数组从数据库中保存的口令字节数组中提取出来，按其长度
        System.arraycopy(pwdInDb, 0, randomByte, 0, BYTE_LENGTH);
        // 创建消息摘要对象
        MessageDigest md = MessageDigest.getInstance("MD5");
        // 将随机数组据传入消息摘要对象
        md.update(randomByte);
        // 将口令的数据传给消息摘要对象
        md.update(password.getBytes("UTF-8"));
        // 生成输入口令的消息摘要
        byte[] digest = md.digest();
        // 声明一个保存数据库中口令消息摘要的变量
        byte[] digestInDb = new byte[pwdInDb.length - BYTE_LENGTH];
        // 取得数据库中口令的消息摘要
        System.arraycopy(pwdInDb, BYTE_LENGTH, digestInDb, 0,digestInDb.length);
        // 比较根据输入口令生成的消息摘要和数据库中消息摘要是否相同
        if (Arrays.equals(digest, digestInDb)) {
            // 口令正确返回口令匹配消息
            return true;
        } else {
            // 口令不正确返回口令不匹配消息
            return false;
        }
    }

    public static void main(String[] args) {

        getMD5to16("asdfasdfa");

    }
}
